# user 角色案例（直接提问）
"""
作用：
    模型直接根据用户的问题生成回答，不受 system 预设身份限制。
    输出更自由，风格完全由模型默认行为决定。
"""
from fastapi import FastAPI
from fastapi import APIRouter, Query
from langchain.schema import HumanMessage, SystemMessage
from langchain_community.llms import Ollama

router = APIRouter()

ollama = Ollama(base_url='http://172.16.21.38:11436', model="qwen3:0.6b")

@router.get("/user_role")
async def user_role():
    # 没有额外的 system 规则，直接是用户提问
    messages = [
        HumanMessage(content="请给我三个具有法国风格的男孩名字")
    ]
    response = ollama.invoke(messages)
    return {"role": "user_demo", "response": response}
